money=5000000
name=input('请输入您的姓名:')

def saving(money,name):
    print('-----存款-----')
    num=int(input('请输入存款金额:'))
    print('%s,您好,您存款%d元成功'%(name,num))
    print('%s,您好,您的余额剩余:%d'%(name,num+money))
    return money+num

def query(name,money):
    print('-----查询余额-----')
    print('%s,您好,您的余额剩余:%d'%(name,money))

def draw(name,money):
    print('-----取款-----')
    num=int(input('请输入取款金额:'))
    print('%s,您好,您取款%d成功'%(name,num))
    print('%s,您好,您的余额剩余:%d元'%(name,money-num))
    return money-num

def mainMenu(name):
    print('-----主菜单-----')
    print('%s,您好,欢迎来到黑马银行ATM.请选择操作:'%name)
    print('查询余额\t[输入1]')
    print('存款\t[输入2]')
    print('取款\t[输入3]')
    print('退出\t[输入4]')
    num=int(input('请输入您的选择:'))

    if num==1:
        query(name,money)
    elif num==2:
        saving(money,name)
    elif num==3:
        draw(name,money)
    return num





num=mainMenu(name)
while num>0 and num<4:
    mainMenu(name)

print('程序已退出')
